#pragma once
#include "DetectPlate.h"
class DetectCover:public DetectPlate
{
public:
	DetectCover();
	~DetectCover();
	int init();

    int get_best_reslut(cv::Mat img, YoloDetection& best_reslut);
    bool findCoverDistance(cv::Mat &img,cv::Rect& rect, float &outx, float &outy, float &outz);
    bool yoloDetect(cv::Mat& img,cv::Rect& rect);
protected:
//	YoloDet detecter;

    std::string modle_path = "chargeRobot/model/charging.onnx";
    std::string class_path = "chargeRobot/model/chargeCover.txt";

    float score_thre = 0.2f;			//类别阈值
    float nms_thre = 0.8f;              //NMS阈值，IOU 大于该值舍去，越大舍去的越少
    float conf_thre = 0.2f;             //置信度
};

